import click
from flask.cli import with_appcontext
from app import db
from app.models import Category, Video, Tag

@click.command('init-db')
@with_appcontext
def init_db_command():
    """初始化数据库"""
    db.create_all()
    click.echo('数据库已初始化.')

@click.command('drop-db')
@with_appcontext
def drop_db_command():
    """删除所有表"""
    if click.confirm('确定要删除所有数据吗？这个操作不可恢复！'):
        db.drop_all()
        click.echo('数据库已清空.')

def init_app(app):
    """注册数据库命令"""
    app.cli.add_command(init_db_command)
    app.cli.add_command(drop_db_command) 